import taichi as ti

ti.init(arch=ti.cuda)


#类内定义field 定义kernel func函数

@ti.data_oriented
class TaichiWhell:
    def __init__(self,radius,width,rolling_fric):
        self.radius = radius
        self.width = width
        self.rolling_fric =rolling_fric
        self.pos=ti.Vector.field(3,ti.f32,shape=4)

    @ti.kernel
    def Roll(self):
        ...
    
    @ti.func
    def foo(self):
        ...
















